--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
Node / ReticulumProjects / MeshChatX.git / files / meshchatx / src / backend / http / routes / websocket_upgrade.py
Displaying Raw • Download
meshchatx/src/backend/http/routes/websocket_upgrade.py 4bd787ccd70a459a02eececca3f942f0e64ec4ce (4bd787cc) Text, 7.81 KB
T8b949e# SPDX-License-Identifier: 0BSD
T8b949e"""HTTP routes: websocket_upgrade."""
Tff7b72from T7ee787__future__ Tff7b72import Te6edf3annotations
Tff7b72from T7ee787meshchatxT7ee787.T7ee787srcT7ee787.T7ee787backendT7ee787.T7ee787httpT7ee787.T7ee787meshchat_names Tff7b72import Tb4b4b4( T8b949e# noqa: F401
Te6edf3GeoValidationErrorTb4b4b4,
Te6edf3OutboundHttpBlockedErrorTb4b4b4,
Te6edf3OverlayExportErrorTb4b4b4,
Te6edf3OverlaySourceParseErrorTb4b4b4,
Te6edf3PluginSecurityErrorTb4b4b4,
Te6edf3AsyncUtilsTb4b4b4,
Te6edf3InterfaceConfigParserTb4b4b4,
Te6edf3InterfaceDiscoveryTb4b4b4,
Te6edf3InterfaceEditorTb4b4b4,
Te6edf3LOGIN_PATHTb4b4b4,
Te6edf3LXMFTb4b4b4,
Te6edf3LxmfAudioFieldTb4b4b4,
Te6edf3LxmfFileAttachmentTb4b4b4,
Te6edf3LxmfFileAttachmentsFieldTb4b4b4,
Te6edf3LxmfImageFieldTb4b4b4,
Te6edf3MAX_EXPORT_TILESTb4b4b4,
Te6edf3MarkdownRendererTb4b4b4,
Te6edf3NomadnetFileDownloaderTb4b4b4,
Te6edf3NomadnetPageDownloaderTb4b4b4,
Te6edf3RNProbeHandlerTb4b4b4,
Te6edf3RNSTb4b4b4,
Te6edf3ReticulumMeshChatTb4b4b4,
Te6edf3SETUP_PATHTb4b4b4,
Te6edf3TRANSPARENT_TILETb4b4b4,
Te6edf3TelemeterTb4b4b4,
Te6edf3UTCTb4b4b4,
Te6edf3WSMsgTypeTb4b4b4,
Te6edf3_is_chaquopy_androidTb4b4b4,
Te6edf3_is_loopback_bind_hostTb4b4b4,
Te6edf3_request_client_ipTb4b4b4,
Te6edf3aiohttpTb4b4b4,
Te6edf3app_versionTb4b4b4,
Te6edf3assert_migration_context_pathsTb4b4b4,
Te6edf3asyncioTb4b4b4,
Te6edf3base64Tb4b4b4,
Te6edf3bcryptTb4b4b4,
Te6edf3binasciiTb4b4b4,
Te6edf3build_blocklist_export_documentTb4b4b4,
Te6edf3build_export_documentTb4b4b4,
Te6edf3build_messages_export_bundleTb4b4b4,
Te6edf3cache_statsTb4b4b4,
Te6edf3cancel_inbound_deliveriesTb4b4b4,
Te6edf3castTb4b4b4,
Te6edf3compute_lxmf_conversation_unread_from_latest_rowTb4b4b4,
Te6edf3configparserTb4b4b4,
Te6edf3contextlibTb4b4b4,
Te6edf3convert_db_favourite_to_dictTb4b4b4,
Te6edf3convert_db_lxmf_message_to_dictTb4b4b4,
Te6edf3convert_lxmf_message_to_dictTb4b4b4,
Te6edf3convert_nomadnet_field_data_to_mapTb4b4b4,
Te6edf3convert_nomadnet_string_data_to_mapTb4b4b4,
Te6edf3convert_propagation_node_state_to_stringTb4b4b4,
Te6edf3copyTb4b4b4,
Te6edf3datetimeTb4b4b4,
Te6edf3describe_port_conflictTb4b4b4,
Te6edf3detect_image_format_from_magicTb4b4b4,
Te6edf3ensure_outbound_http_allowedTb4b4b4,
Te6edf3ensure_session_csrf_tokenTb4b4b4,
Te6edf3filter_announced_dicts_by_search_queryTb4b4b4,
Te6edf3fresh_storage_at_targetTb4b4b4,
Te6edf3get_cached_active_linkTb4b4b4,
Te6edf3get_file_pathTb4b4b4,
Te6edf3get_sessionTb4b4b4,
Te6edf3get_trusted_proxy_cidrsTb4b4b4,
Te6edf3gif_utilsTb4b4b4,
Te6edf3i2p_supportTb4b4b4,
Te6edf3import_messages_export_bundleTb4b4b4,
Te6edf3ioTb4b4b4,
Te6edf3is_mbtiles_filenameTb4b4b4,
Te6edf3is_path_within_dirTb4b4b4,
Te6edf3is_port_in_useTb4b4b4,
Te6edf3is_user_facing_lxmf_payloadTb4b4b4,
Te6edf3jsonTb4b4b4,
Te6edf3list_host_network_interfacesTb4b4b4,
Te6edf3list_inbound_deliveriesTb4b4b4,
Te6edf3list_portsTb4b4b4,
Te6edf3load_app_security_settingsTb4b4b4,
Te6edf3loggerTb4b4b4,
Te6edf3loggingTb4b4b4,
Te6edf3lxmf_sidebar_preview_for_conversation_latest_rowTb4b4b4,
Te6edf3memory_log_handlerTb4b4b4,
Te6edf3message_fields_have_attachmentsTb4b4b4,
Te6edf3migrate_legacy_to_targetTb4b4b4,
Te6edf3mime_for_image_typeTb4b4b4,
Te6edf3normalize_identity_storage_hashTb4b4b4,
Te6edf3normalize_lxmf_sieve_filtersTb4b4b4,
Te6edf3normalize_message_blocklistTb4b4b4,
Te6edf3osTb4b4b4,
Te6edf3parse_bool_query_paramTb4b4b4,
Te6edf3parse_import_documentTb4b4b4,
Te6edf3parse_lxmf_display_nameTb4b4b4,
Te6edf3parse_lxmf_propagation_node_app_dataTb4b4b4,
Te6edf3parse_lxmf_sieve_filters_jsonTb4b4b4,
Te6edf3parse_lxmf_stamp_costTb4b4b4,
Te6edf3parse_message_blocklist_jsonTb4b4b4,
Te6edf3parse_nomadnetwork_node_display_nameTb4b4b4,
Te6edf3platformTb4b4b4,
Te6edf3privacy_mode_enabledTb4b4b4,
Te6edf3psutilTb4b4b4,
Te6edf3purge_messages_before_cutoffTb4b4b4,
Te6edf3reTb4b4b4,
Te6edf3resolve_message_age_cutoffTb4b4b4,
Te6edf3reticulum_pathfindingTb4b4b4,
Te6edf3rotate_session_csrf_tokenTb4b4b4,
Te6edf3rrc_protocolTb4b4b4,
Te6edf3safe_path_under_dirTb4b4b4,
Te6edf3sanitize_sticker_emojiTb4b4b4,
Te6edf3sanitize_sticker_nameTb4b4b4,
Te6edf3sanitize_websocket_config_updateTb4b4b4,
Te6edf3save_app_security_settingsTb4b4b4,
Te6edf3secretsTb4b4b4,
Te6edf3shutilTb4b4b4,
Te6edf3sqlite3Tb4b4b4,
Te6edf3sticker_pack_utilsTb4b4b4,
Te6edf3sysTb4b4b4,
Te6edf3tempfileTb4b4b4,
Te6edf3threadingTb4b4b4,
Te6edf3timeTb4b4b4,
Te6edf3tracebackTb4b4b4,
Te6edf3user_agent_hashTb4b4b4,
Te6edf3validate_export_documentTb4b4b4,
Te6edf3webTb4b4b4,
Te6edf3websocket_type_requires_authTb4b4b4,
Te6edf3zipfileTb4b4b4,
Tb4b4b4)
Tff7b72def Td2a8ffregister_websocket_upgrade_routesTb4b4b4(Te6edf3routesTb4b4b4, Te6edf3appTb4b4b4)Tb4b4b4:
T8b949e# handle websocket clients
Tf0883e@routesTff7b72.Td2a8ffgetTb4b4b4(Ta5d6ff"Ta5d6ff/wsTa5d6ff"Tb4b4b4)
Tff7b72async Tff7b72def Td2a8ffwsTb4b4b4(Te6edf3requestTb4b4b4)Tb4b4b4:
T8b949e# prepare websocket response
Te6edf3websocket_response Tff7b72= Te6edf3webTff7b72.Td2a8ffWebSocketResponseTb4b4b4(
T8b949e# set max message size accepted by server to 50 megabytes
Te6edf3max_msg_sizeTff7b72=T79c0ff50 Tff7b72* T79c0ff1024 Tff7b72* T79c0ff1024Tb4b4b4,
Tb4b4b4)
Tff7b72await Te6edf3websocket_responseTff7b72.Td2a8ffprepareTb4b4b4(Te6edf3requestTb4b4b4)
T8b949e# aiohttp WebSocketResponse does not expose .request, so keep it for
T8b949e# session checks on authenticated mutators (nomadnet downloads, etc).
Te6edf3websocket_responseTff7b72.Td2a8ff_meshchatx_request Tff7b72= Te6edf3request
T8b949e# add client to connected clients list
Te6edf3appTff7b72.Td2a8ffwebsocket_clientsTff7b72.Td2a8ffappendTb4b4b4(Te6edf3websocket_responseTb4b4b4)
Te6edf3session Tff7b72= Te6edf3appTff7b72.Td2a8ffactive_sessionsTff7b72.Td2a8ffaddTb4b4b4(
Te6edf3ipTff7b72=Te6edf3requestTff7b72.Td2a8ffremoteTb4b4b4,
Te6edf3user_agentTff7b72=Te6edf3requestTff7b72.Td2a8ffheadersTff7b72.Td2a8ffgetTb4b4b4(Ta5d6ff"Ta5d6ffUser-AgentTa5d6ff"Tb4b4b4)Tb4b4b4,
Tb4b4b4)
Te6edf3websocket_responseTff7b72.Td2a8ff_meshchatx_session_id Tff7b72= Te6edf3sessionTb4b4b4[Ta5d6ff"Ta5d6ffidTa5d6ff"Tb4b4b4]
T8b949e# send config to all clients
Tff7b72await Te6edf3appTff7b72.Td2a8ffsend_config_to_websocket_clientsTb4b4b4(Tb4b4b4)
Tff7b72await Te6edf3appTff7b72.Td2a8ffsend_active_sessions_to_websocket_clientsTb4b4b4(Tb4b4b4)
T8b949e# handle websocket messages until disconnected
Tff7b72async Tff7b72for Te6edf3msg Tff7b72in Te6edf3websocket_responseTb4b4b4:
Te6edf3message Tff7b72= Te6edf3castTb4b4b4(Ta5d6ff"Ta5d6ffWSMessageTa5d6ff"Tb4b4b4, Te6edf3msgTb4b4b4)
Tff7b72if Te6edf3messageTff7b72.Td2a8fftype Tff7b72== Te6edf3WSMsgTypeTff7b72.Td2a8ffTEXTTb4b4b4:
Tff7b72tryTb4b4b4:
Te6edf3data Tff7b72= Te6edf3jsonTff7b72.Td2a8ffloadsTb4b4b4(Te6edf3messageTff7b72.Td2a8ffdataTb4b4b4)
Tff7b72await Te6edf3appTff7b72.Td2a8ffon_websocket_data_receivedTb4b4b4(Te6edf3websocket_responseTb4b4b4, Te6edf3dataTb4b4b4)
Tff7b72except Tf85149Exception Tff7b72as Te6edf3eTb4b4b4:
T8b949e# ignore errors while handling message
Tffa657printTb4b4b4(Ta5d6ff"Ta5d6fffailed to process client messageTa5d6ff"Tb4b4b4)
Tffa657printTb4b4b4(Te6edf3eTb4b4b4)
Tff7b72elif Te6edf3messageTff7b72.Td2a8fftype Tff7b72== Te6edf3WSMsgTypeTff7b72.Td2a8ffERRORTb4b4b4:
T8b949e# ignore errors while handling message
Tffa657printTb4b4b4(Ta5d6fffTa5d6ff"Ta5d6ffws connection error Tffd700{Te6edf3websocket_responseTff7b72.Td2a8ffexceptionTb4b4b4(Tb4b4b4)Tffd700}Ta5d6ff"Tb4b4b4)
T8b949e# websocket closed
Tff7b72tryTb4b4b4:
Te6edf3appTff7b72.Td2a8ffwebsocket_clientsTff7b72.Td2a8ffremoveTb4b4b4(Te6edf3websocket_responseTb4b4b4)
Tff7b72except Tf85149ValueErrorTb4b4b4:
Tff7b72pass
Te6edf3appTff7b72.Td2a8ff_detach_active_sessionTb4b4b4(Te6edf3websocket_responseTb4b4b4)
Te6edf3appTff7b72.Td2a8ff_cancel_rns_link_tasks_for_clientTb4b4b4(Te6edf3websocket_responseTb4b4b4)
Tff7b72await Te6edf3appTff7b72.Td2a8ffsend_active_sessions_to_websocket_clientsTb4b4b4(Tb4b4b4)
Tff7b72return Te6edf3websocket_response
Tf0883e@routesTff7b72.Td2a8ffgetTb4b4b4(Ta5d6ff"Ta5d6ff/ws/telephone/audioTa5d6ff"Tb4b4b4)
Tff7b72async Tff7b72def Td2a8fftelephone_audio_wsTb4b4b4(Te6edf3requestTb4b4b4)Tb4b4b4:
Te6edf3websocket_response Tff7b72= Te6edf3webTff7b72.Td2a8ffWebSocketResponseTb4b4b4(
T8b949e# Cap well above a normal PCM frame (tens of KB) but far below prior 5 MiB.
Te6edf3max_msg_sizeTff7b72=T79c0ff256 Tff7b72* T79c0ff1024Tb4b4b4,
Tb4b4b4)
Tff7b72await Te6edf3websocket_responseTff7b72.Td2a8ffprepareTb4b4b4(Te6edf3requestTb4b4b4)
T8b949e# Chaquopy Android and headless/web deployments have no usable LXST
T8b949e# host audio device, so always allow the websocket bridge.
Te6edf3web_audio_allowed Tff7b72= Tb4b4b4(
Te6edf3appTff7b72.Td2a8ffweb_audio_bridgeTff7b72.Td2a8ffconfig_enabledTb4b4b4(Tb4b4b4) Tff7b72or Te6edf3appTff7b72.Td2a8ffweb_audio_requiredTb4b4b4(Tb4b4b4)
Tb4b4b4)
Tff7b72if Tff7b72not Te6edf3web_audio_allowedTb4b4b4:
Tff7b72await Te6edf3websocket_responseTff7b72.Td2a8ffsend_strTb4b4b4(
Te6edf3jsonTff7b72.Td2a8ffdumpsTb4b4b4(
Tb4b4b4{Ta5d6ff"Ta5d6fftypeTa5d6ff"Tb4b4b4: Ta5d6ff"Ta5d6fferrorTa5d6ff"Tb4b4b4, Ta5d6ff"Ta5d6ffmessageTa5d6ff"Tb4b4b4: Ta5d6ff"Ta5d6ffWeb audio is disabled in configTa5d6ff"Tb4b4b4}Tb4b4b4,
Tb4b4b4)Tb4b4b4,
Tb4b4b4)
Tff7b72await Te6edf3websocket_responseTff7b72.Td2a8ffcloseTb4b4b4(Tb4b4b4)
Tff7b72return Te6edf3websocket_response
Tff7b72await Te6edf3appTff7b72.Td2a8ffweb_audio_bridgeTff7b72.Td2a8ffsend_statusTb4b4b4(Te6edf3websocket_responseTb4b4b4)
Te6edf3attached Tff7b72= Te6edf3appTff7b72.Td2a8ffweb_audio_bridgeTff7b72.Td2a8ffattach_clientTb4b4b4(Te6edf3websocket_responseTb4b4b4)
Tff7b72if Tff7b72not Te6edf3attachedTb4b4b4:
Tff7b72await Te6edf3websocket_responseTff7b72.Td2a8ffsend_strTb4b4b4(
Te6edf3jsonTff7b72.Td2a8ffdumpsTb4b4b4(
Tb4b4b4{Ta5d6ff"Ta5d6fftypeTa5d6ff"Tb4b4b4: Ta5d6ff"Ta5d6fferrorTa5d6ff"Tb4b4b4, Ta5d6ff"Ta5d6ffmessageTa5d6ff"Tb4b4b4: Ta5d6ff"Ta5d6ffNo active call to attachTa5d6ff"Tb4b4b4}Tb4b4b4,
Tb4b4b4)Tb4b4b4,
Tb4b4b4)
Tff7b72async Tff7b72for Te6edf3msg Tff7b72in Te6edf3websocket_responseTb4b4b4:
Te6edf3message Tff7b72= Te6edf3castTb4b4b4(Ta5d6ff"Ta5d6ffWSMessageTa5d6ff"Tb4b4b4, Te6edf3msgTb4b4b4)
Tff7b72if Te6edf3messageTff7b72.Td2a8fftype Tff7b72== Te6edf3WSMsgTypeTff7b72.Td2a8ffBINARYTb4b4b4:
T8b949e# Only accept PCM after a successful attach for this socket.
Tff7b72if Te6edf3websocket_response Tff7b72in Te6edf3appTff7b72.Td2a8ffweb_audio_bridgeTff7b72.Td2a8ffclientsTb4b4b4:
Te6edf3appTff7b72.Td2a8ffweb_audio_bridgeTff7b72.Td2a8ffpush_client_frameTb4b4b4(Te6edf3messageTff7b72.Td2a8ffdataTb4b4b4)
Tff7b72elif Te6edf3messageTff7b72.Td2a8fftype Tff7b72== Te6edf3WSMsgTypeTff7b72.Td2a8ffTEXTTb4b4b4:
Tff7b72tryTb4b4b4:
Te6edf3data Tff7b72= Te6edf3jsonTff7b72.Td2a8ffloadsTb4b4b4(Te6edf3messageTff7b72.Td2a8ffdataTb4b4b4)
Tff7b72if Te6edf3dataTff7b72.Td2a8ffgetTb4b4b4(Ta5d6ff"Ta5d6fftypeTa5d6ff"Tb4b4b4) Tff7b72== Ta5d6ff"Ta5d6ffattachTa5d6ff"Tb4b4b4:
Te6edf3appTff7b72.Td2a8ffweb_audio_bridgeTff7b72.Td2a8ffattach_clientTb4b4b4(Te6edf3websocket_responseTb4b4b4)
Tff7b72elif Te6edf3dataTff7b72.Td2a8ffgetTb4b4b4(Ta5d6ff"Ta5d6fftypeTa5d6ff"Tb4b4b4) Tff7b72== Ta5d6ff"Ta5d6ffpingTa5d6ff"Tb4b4b4:
Tff7b72await Te6edf3websocket_responseTff7b72.Td2a8ffsend_strTb4b4b4(
Te6edf3jsonTff7b72.Td2a8ffdumpsTb4b4b4(Tb4b4b4{Ta5d6ff"Ta5d6fftypeTa5d6ff"Tb4b4b4: Ta5d6ff"Ta5d6ffpongTa5d6ff"Tb4b4b4}Tb4b4b4)Tb4b4b4,
Tb4b4b4)
Tff7b72except Tf85149Exception Tff7b72as Te6edf3eTb4b4b4:
Te6edf3loggingTff7b72.Td2a8ffexceptionTb4b4b4(
Ta5d6fffTa5d6ff"Ta5d6ffError processing websocket text message: Tffd700{Te6edf3eTffd700}Ta5d6ff"Tb4b4b4,
Tb4b4b4)
Tff7b72elif Te6edf3messageTff7b72.Td2a8fftype Tff7b72== Te6edf3WSMsgTypeTff7b72.Td2a8ffERRORTb4b4b4:
Tffa657printTb4b4b4(Ta5d6fffTa5d6ff"Ta5d6fftelephone audio ws error Tffd700{Te6edf3websocket_responseTff7b72.Td2a8ffexceptionTb4b4b4(Tb4b4b4)Tffd700}Ta5d6ff"Tb4b4b4)
Te6edf3appTff7b72.Td2a8ffweb_audio_bridgeTff7b72.Td2a8ffdetach_clientTb4b4b4(Te6edf3websocket_responseTb4b4b4)
Tff7b72return Te6edf3websocket_response
──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────